package LeetCode1;

import java.util.HashMap;
import java.util.Map;
import java.util.Stack;

//下一个更大元素！
//nums1 中数字 x 的 下一个更大元素 是指 x 在 nums2 中对应位置 右侧 的 第一个 比 x 大的元素。
//给你两个 没有重复元素 的数组 nums1 和 nums2 ，下标从 0 开始计数，其中nums1 是 nums2 的子集
//返回一个长度为 nums1.length 的数组 ans 作为答案，满足 ans[i] 是如上所述的 下一个更大元素 。
public class Num496 {
    //单调栈，栈顶到栈底从小到大
    private static Stack<Integer> stack = new Stack<>();
    //存储nums2中每个元素，以及它的下一个比他大的元素值
    private static Map<Integer, Integer> map = new HashMap<>();

    public static void main(String[] args) {
        int[] a = {4,1,2};
        int[] b = {1,3,4,2};
        int[] c = nextGreaterElement(a,b);
        for (int i = 0; i < c.length; i++) {
            System.out.println(c[i]);
        }
    }
    public static int[] nextGreaterElement(int[] nums1, int[] nums2) {
        //单调栈解决
        //1.扫描nums2中的元素，维护单调栈，找到大map存储元素
        for (int i = nums2.length - 1; i >= 0; i--) {
            //当栈不为空时，将比当前元素小的栈顶元素出栈
            while (!stack.isEmpty() && stack.peek() <= nums2[i]) {
                stack.pop();
            }
            //将比当前元素大的栈顶元素记录
            int nextGreaterValue = stack.isEmpty() ? -1 : stack.peek();
            //将当前元素入栈
            stack.push(nums2[i]);
            //记录当前元素，以及元素的下一大值
            map.put(nums2[i], nextGreaterValue);
        }
        int[] ret = new int[nums1.length];
        for (int i = 0; i < nums1.length; i++) {
            ret[i] = map.get(nums1[i]);
        }
        return ret;
        //for循环加数组
//        int[] arr = new int[nums1.length];
//        for (int i = 0; i < nums1.length; i++) {
//            for (int j = 0; j < nums2.length; j++) {
//                if (nums1[i] == nums2[j])
//                    arr[i] = j;
//            }
//        }
//        int[] brr = new int[nums1.length];
//        for (int i = 0; i < nums1.length; i++) {
//            brr[i] = -1;
//        }
//        for (int i = 0; i < nums1.length; i++) {
//            for (int j = arr[i]; j < nums2.length; j++) {
//                if (nums2[j] > nums1[i]) {
//                    brr[i] = nums2[j];
//                    break;
//                }
//            }
//        }
//        return brr;
    }
}
